package com.heima.media.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.heima.common.dto.PageResponseResult;
import com.heima.common.dto.ResponseResult;
import com.heima.common.dto.User;
import com.heima.common.enums.AppHttpCodeEnum;
import com.heima.common.util.MediaThreadLocalUtil;
import com.heima.media.dto.ContentDto;
import com.heima.media.dto.ImageDto;
import com.heima.media.dto.WmNewsDto;
import com.heima.media.dto.WmNewsPageDto;
import com.heima.media.entity.WmNews;
import com.heima.media.entity.WmNewsMaterial;
import com.heima.media.mapper.WmNewsMapper;
import com.heima.media.service.IAuditService;
import com.heima.media.service.IWmNewsMaterialService;
import com.heima.media.service.IWmNewsService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Lazy;
import org.springframework.kafka.core.KafkaTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * <p>
 * 自媒体图文内容信息表 服务实现类
 * </p>
 *
 * @author mcm
 * @since 2022-04-12
 */
@Service
public class WmNewsServiceImpl extends ServiceImpl<WmNewsMapper, WmNews> implements IWmNewsService {

    @Override
    public ResponseResult listByCondition(WmNewsPageDto dto) {
        // 查询条件
        LambdaQueryWrapper<WmNews> query = new LambdaQueryWrapper<>();
        // 判断当前登录的用户
        // 从本地线程查询用户
        User user = MediaThreadLocalUtil.get();
        if (user == null) {
            return ResponseResult.errorResult(AppHttpCodeEnum.NEED_LOGIN);
        }
        query.eq(WmNews::getUserId, user.getUserId());
        // 判断状态是否为空
        if (dto.getStatus() != null) {
            query.eq(WmNews::getStatus, dto.getStatus());
        }
        // 判断关键字是否为空
        if (!StringUtils.isEmpty(dto.getKeyword())) {
            query.like(WmNews::getTitle, dto.getKeyword());
        }
        // 判断频道id
        if (dto.getChannelId() != null) {
            query.eq(WmNews::getChannelId, dto.getChannelId());
        }
        // 判断发布日期
        if (dto.getBeginPubDate() != null) {
            query.ge(WmNews::getPublishTime, dto.getBeginPubDate());
        }
        if (dto.getEndPubDate() != null) {
            query.lt(WmNews::getPublishTime, dto.getEndPubDate());
        }
        // 按照发布时间排序
        query.orderByDesc(WmNews::getPublishTime);
        // 分页查询
        IPage<WmNews> page = new Page<>(dto.getPage(), dto.getSize());

        IPage<WmNews> iPage = this.page(page, query);
        PageResponseResult result = new PageResponseResult(dto.getPage(), dto.getSize(), iPage.getTotal(), iPage.getRecords());
        return result;
    }

    @Override
    public ResponseResult submit(WmNewsDto dto) {

        // 从本地线程查询用户
        User user = MediaThreadLocalUtil.get();
        if (user == null) {
            return ResponseResult.errorResult(AppHttpCodeEnum.NEED_LOGIN);
        }

        // 判断dto是否包含文章id
        // 有id,是更新文章,需要先删除已有文章-素材关联关系
        if (dto.getId() != null) {
            deleteRelation(dto.getId());
        }

        // 没有id,新增文章
        WmNews wmNews = new WmNews();
        BeanUtils.copyProperties(dto, wmNews);
        wmNews.setUserId(user.getUserId());
        wmNews.setCreatedTime(new Date());
        wmNews.setSubmitedTime(new Date());
        wmNews.setImages(JSON.toJSONString(dto.getImages()));
        // 从前端传递的内容中提取图片
        List<ImageDto> contentImages = getImagesFromContent(wmNews.getContent());
        if (dto.getType() == -1) {
            // 需要从内容中自动提取图片作为封面图片
            // 判断内容图片的个数
            if (contentImages.size() <= 0) {
                // 修改类型为无图
                wmNews.setType(0);
            } else if (contentImages.size() < 3) {
                // 修改类型为单图
                wmNews.setType(1);
                // 从内容图片中取第一张图片作为封面
                List<ImageDto> imageList = contentImages.subList(0, 1);
                wmNews.setImages(JSON.toJSONString(imageList));
            } else {
                // 修改类型为多图
                wmNews.setType(3);
                // 从内容图片中取前3张图片作为封面
                List<ImageDto> imageList = contentImages.subList(0, 3);
                wmNews.setImages(JSON.toJSONString(imageList));
            }
        }

        // 保存文章
        // this.save(wmNews);
        this.saveOrUpdate(wmNews);
        // 判断是否为草稿
        if (dto.getStatus() != 0) {
            // 不为草稿,保存文章-素材关联关系  包含内容关联和封面关联
            // 添加封面关联
            List<ImageDto> coverImages = JSON.parseArray(wmNews.getImages(), ImageDto.class);
            saveRelation(coverImages, wmNews.getId(), 1);

            // 添加内容关联
            saveRelation(contentImages, wmNews.getId(), 0);

            // 调用自动审核服务
            System.out.println(Thread.currentThread().getName() + " 开始调用自动审核服务...");
            auditService.audit(wmNews);
        }
        return ResponseResult.okResult();
    }

    @Override
    public ResponseResult downOrUp(WmNewsDto dto) {
        WmNews wmNews = this.getById(dto.getId());
        // 修改文章的上下架状态
        wmNews.setEnable(dto.getEnable() == 1 ? true : false);
        this.updateById(wmNews);
        // 同步状态到文章库
        // 使用kafka来发送消息
        String key = wmNews.getArticleId() + "";
        String value = dto.getEnable() == 1 ? "false" : "true";
        kafkaTemplate.send(upDownTopic, key, value);
        return ResponseResult.okResult();
    }

    @Value("${topic.upDownTopic}")
    private String upDownTopic;

    @Autowired
    private KafkaTemplate<String, String> kafkaTemplate;

    @Autowired
    @Lazy
    private IAuditService auditService;

    /**
     * 删除文章-素材关联
     *
     * @param id
     */
    private void deleteRelation(Integer id) {
        LambdaQueryWrapper<WmNewsMaterial> query = new LambdaQueryWrapper<>();
        query.eq(WmNewsMaterial::getNewsId, id);
        newsMaterialService.remove(query);
    }

    /**
     * 从前端传递的内容中提取图片
     *
     * @param content
     * @return
     */
    private List<ImageDto> getImagesFromContent(String content) {
        List<ImageDto> contentImages = new ArrayList<>();
        List<ContentDto> dtos = JSON.parseArray(content, ContentDto.class);
        for (ContentDto dto : dtos) {
            if (dto.getType().equals("image")) {
                String url = dto.getValue();
                Integer id = dto.getId();
                ImageDto imageDto = new ImageDto(id, url);
                contentImages.add(imageDto);
            }
        }
        return contentImages;
    }

    @Autowired
    private IWmNewsMaterialService newsMaterialService;

    /**
     * 保存关联关系
     *
     * @param images
     * @param newsId
     * @param type
     */
    private void saveRelation(List<ImageDto> images, Integer newsId, int type) {
        int ord = 0;
        for (ImageDto dto : images) {
            WmNewsMaterial entity = new WmNewsMaterial();
            entity.setMaterialId(dto.getId());
            entity.setNewsId(newsId);
            entity.setType(type);
            entity.setOrd(ord);
            newsMaterialService.save(entity);
            ord++;
        }
    }
}
